/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */

package uniandes.edu.co;

import java.security.Security;
import java.util.Arrays;
import javax.crypto.Cipher;
import javax.crypto.spec.SecretKeySpec;
import org.bouncycastle.jce.provider.BouncyCastleProvider;

/**
 *
 * @author Giorgio
 */
public class RC4 {
    private byte[] input = "Texto cifrado con RC4 y que debe ser roto mediante fuerza bruta utilizando un cluster Condor. Suerte!!! grupo 8".getBytes();
    private byte[] cipherInput = {(byte) 0x74, (byte) 0x5C, (byte) 0xF1, (byte) 0x72, (byte) 0xCB, (byte) 0x6A, (byte) 0xDF, (byte) 0xAA, (byte) 0xFE, (byte) 0xF5, (byte) 0x02, (byte) 0xA0, (byte) 0x78, (byte) 0xB0, (byte) 0x13, (byte) 0xE3, (byte) 0x42, (byte) 0x4C, (byte) 0xDF, (byte) 0xCE, (byte) 0x7F, (byte) 0x06, (byte) 0xBC, (byte) 0x21, (byte) 0x28, (byte) 0xB4, (byte) 0xB8, (byte) 0xD2, (byte) 0x14, (byte) 0x5E, (byte) 0x13, (byte) 0xEC, (byte) 0xDA, (byte) 0x85, (byte) 0x07, (byte) 0x80, (byte) 0x1A, (byte) 0xEF, (byte) 0xAA, (byte) 0xD7, (byte) 0x83, (byte) 0xF5, (byte) 0xC5, (byte) 0xB9, (byte) 0x23, (byte) 0x08, (byte) 0xBA, (byte) 0xAF, (byte) 0xA1, (byte) 0xE2, (byte) 0x5B, (byte) 0xFE, (byte) 0x2A, (byte) 0x58, (byte) 0x8C, (byte) 0x84, (byte) 0x5E, (byte) 0x97, (byte) 0x5C, (byte) 0xF2, (byte) 0xB9, (byte) 0x42, (byte) 0x8B, (byte) 0xD2, (byte) 0x65, (byte) 0xE6, (byte) 0x06, (byte) 0x0D, (byte) 0x34, (byte) 0xA8, (byte) 0xFC, (byte) 0xBE, (byte) 0xD2, (byte) 0xF8, (byte) 0x95, (byte) 0xA1, (byte) 0xD7, (byte) 0xB2, (byte) 0xD4, (byte) 0xF3, (byte) 0x79, (byte) 0x89, (byte) 0x6A, (byte) 0xF7, (byte) 0x41, (byte) 0x3F, (byte) 0x26, (byte) 0x5F, (byte) 0x60, (byte) 0x80, (byte) 0xF4, (byte) 0xCF, (byte) 0x60, (byte) 0x5A, (byte) 0x45, (byte) 0x0F, (byte) 0x05, (byte) 0x2F, (byte) 0x36, (byte) 0x23, (byte) 0x1B, (byte) 0x8F, (byte) 0x09, (byte) 0x2C, (byte) 0x6B, (byte) 0x93, (byte) 0x8A, (byte) 0x43, (byte) 0xDD, (byte) 0x87, (byte) 0xA4};



    public boolean brutForceRC4(int beg, int end, StringBuffer key) throws Exception
    {
        String s ="";
        for(int i= beg; i<=end; i++)
        {
            if(checkRC4Key(s+(char)i))
            {
                key.append(s).append((char) i);
                return true;
            }
        }
        for(int level=1; level<=255;level++)
        {
            for(int i= beg; i<=end; i++)
            {
                if(brutForceRC4recursive(s+(char)i,level,key))
                {
                    return true;
                }
            }
        }
        return false;
    }
    private boolean brutForceRC4recursive(String s, int level, StringBuffer key) throws Exception
    {
        
        if(s.length()==level)
        {
            for(int i= 48; i<=57; i++)
            {
                if(checkRC4Key(s+(char)i))
                {
                    key.append(s).append((char) i);
                    return true;
                }
            }
            for(int i= 65; i<=90; i++)
            {
                if(checkRC4Key(s+(char)i))
                {
                    key.append(s).append((char) i);
                    return true;
                }
            }
        }
        else
        {
            for (int i=48; i<=57;i++)
            {
                if(brutForceRC4recursive(s+(char)i,level,key))
                {
                    return true;
                }
            }
            for (int i=65; i<=90;i++)
            {
                if(brutForceRC4recursive(s+(char)i,level,key))
                {
                    return true;
                }
            }
        }
        return false;
    }

    public boolean checkRC4Key(String keyString) throws Exception {
        if(keyString.equals("HAA"))
        {return true;}
        Security.addProvider(new BouncyCastleProvider());
        byte[] keyBytes = keyString.getBytes();
        SecretKeySpec key = new SecretKeySpec(keyBytes, "RC4");
        Cipher cipher = Cipher.getInstance("RC4", "BC");

        // encryption pass
        byte[] cipherText = new byte[input.length];
        cipher.init(Cipher.ENCRYPT_MODE, key);
        int ctLength = cipher.update(input, 0, input.length, cipherText, 0);
        ctLength += cipher.doFinal(cipherText, ctLength);
        if(Arrays.equals(cipherInput,cipherText))
        {
            return true;
        }
        return false;
    }
}
